/*
 * Copyright 2007 Google Inc.
 * 
 * Licensed under the Apache License, Version 2.0 (the "License"); you may not
 * use this file except in compliance with the License. You may obtain a copy of
 * the License at
 * 
 * http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
 * License for the specific language governing permissions and limitations under
 * the License.
 */
package com.google.gwt.dev.jjs.ast.js;

import com.google.gwt.core.ext.js.JsniFunction;
import com.google.gwt.dev.jjs.SourceInfo;
import com.google.gwt.dev.jjs.ast.Context;
import com.google.gwt.dev.jjs.ast.JAbstractMethodBody;
import com.google.gwt.dev.jjs.ast.JProgram;
import com.google.gwt.dev.jjs.ast.JVisitor;

import java.util.List;
import java.util.Set;

/**
 * Represents a the body of a method. Can be Java or JSNI.
 */
public class JsniMethodBody extends JAbstractMethodBody {

  private JsniFunction jsFunction = null;

  private String jsSource = null;

  public JsniMethodBody(JProgram program, SourceInfo info, String jsSource) {
    super(program, info);
    this.jsSource = jsSource;
  }

  public JsniFunction getFunc() {
    assert (this.jsFunction != null);
    return jsFunction;
  }

  public String getSource() {
    return jsSource;
  }

  public Set<String> getUsedStrings() {
    return jsFunction.getStringLiterals();
  }

  @Override
  public boolean isNative() {
    return true;
  }

  public void setFunc(JsniFunction func) {
    assert (jsFunction == null);
    this.jsFunction = func;
  }

  public void setSource(String jsSrc) {
    this.jsSource = jsSrc;
  }

  @SuppressWarnings("unchecked")
  public void traverse(JVisitor visitor, Context ctx) {
    if (visitor.visit(this, ctx)) {
      visitor.accept((List) jsFunction.getJavaRefs());
    }
    visitor.endVisit(this, ctx);
  }
}
